home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-9.10-netbook-remix-PL.iso / casper / filesystem.squashfs / usr / share / hplip / ui4 / plugindialog.py < prev    next >
Encoding:
Python Source  |  2009-10-09  |  15.1 KB  |  379 lines
  1. # -*- coding: utf-8 -*-
  2. #
  3. # (c) Copyright 2001-2009 Hewlett-Packard Development Company, L.P.
  4. #
  5. # This program is free software; you can redistribute it and/or modify
  6. # it under the terms of the GNU General Public License as published by
  7. # the Free Software Foundation; either version 2 of the License, or
  8. # (at your option) any later version.
  9. #
  10. # This program is distributed in the hope that it will be useful,
  11. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13. # GNU General Public License for more details.
  14. #
  15. # You should have received a copy of the GNU General Public License
  16. # along with this program; if not, write to the Free Software
  17. # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  18. #
  19. # Authors: Don Welch
  20. #
  21.  
  22.  
  23. # Local
  24. from base.g import *
  25. from base import device, utils
  26. from prnt import cups
  27. from base.codes import *
  28. from ui_utils import *
  29. from installer.core_install import CoreInstall
  30. from installer.core_install import  PLUGIN_INSTALL_ERROR_NONE, \
  31.                                     PLUGIN_INSTALL_ERROR_PLUGIN_FILE_NOT_FOUND, \
  32.                                     PLUGIN_INSTALL_ERROR_DIGITAL_SIG_NOT_FOUND, \
  33.                                     PLUGIN_INSTALL_ERROR_DIGITAL_SIG_BAD, \
  34.                                     PLUGIN_INSTALL_ERROR_PLUGIN_FILE_CHECKSUM_ERROR, \
  35.                                     PLUGIN_INSTALL_ERROR_NO_NETWORK, \
  36.                                     PLUGIN_INSTALL_ERROR_DIRECTORY_ERROR, \
  37.                                     PLUGIN_INSTALL_ERROR_UNABLE_TO_RECV_KEYS
  38.  
  39. # Qt
  40. from PyQt4.QtCore import *
  41. from PyQt4.QtGui import *
  42.  
  43. # Ui
  44. from plugindialog_base import Ui_Dialog
  45.  
  46.  
  47. PAGE_SOURCE = 0
  48. # PAGE_LICENSE = 1 # part of plug-in itself, this is a placeholder
  49. PAGE_MAX = 1
  50.  
  51.  
  52.  
  53. class PluginDialog(QDialog, Ui_Dialog):
  54.     def __init__(self, parent, install_mode=PLUGIN_NONE, plugin_reason=PLUGIN_REASON_NONE):
  55.         QDialog.__init__(self, parent)
  56.         self.install_mode = install_mode
  57.         self.plugin_reason = plugin_reason
  58.         self.plugin_path = None
  59.         self.result = False
  60.         self.core = CoreInstall()
  61.         self.core.set_plugin_version()
  62.         self.setupUi(self)
  63.  
  64.         self.user_settings = UserSettings()
  65.         self.user_settings.load()
  66.         self.user_settings.debug()
  67.  
  68.         self.initUi()
  69.  
  70.         QTimer.singleShot(0, self.showSourcePage)
  71.  
  72.  
  73.     def isPluginInstalled(self):
  74.         return self.core.check_for_plugin()
  75.  
  76.  
  77.     def initUi(self):
  78.         # connect signals/slots
  79.         self.connect(self.CancelButton, SIGNAL("clicked()"), self.CancelButton_clicked)
  80.         self.connect(self.NextButton, SIGNAL("clicked()"), self.NextButton_clicked)
  81.  
  82.         # Application icon
  83.         self.setWindowIcon(QIcon(load_pixmap('hp_logo', '128x128')))
  84.  
  85.         self.PLUGIN_REASON_TEXT = {
  86.             PLUGIN_REASON_NONE: None,
  87.             PLUGIN_REASON_PRINTING_SUPPORT: self.__tr("This plugin will enable printing support."),
  88.             PLUGIN_REASON_FASTER_PRINTING: self.__tr("This plugin will enhance print speed."),
  89.             PLUGIN_REASON_BETTER_PRINTING_PQ: self.__tr("This plugin will enhance print quality."),
  90.             PLUGIN_REASON_PRINTING_FEATURES: self.__tr("This plugin will add printing features."),
  91.             PLUGIN_REASON_RESERVED_10: None,
  92.             PLUGIN_REASON_RESERVED_20: None,
  93.             PLUGIN_REASON_SCANNING_SUPPORT: self.__tr("This plugin will enable scanning support."),
  94.             PLUGIN_REASON_FASTER_SCANNING: self.__tr("This plugin will enhance scanning speed."),
  95.             PLUGIN_REASON_BETTER_SCANNING_IQ: self.__tr("This plugin will enhance scanning image quality."),
  96.             PLUGIN_REASON_RESERVED_200: None,
  97.             PLUGIN_REASON_RESERVED_400: None,
  98.             PLUGIN_REASON_FAXING_SUPPORT: self.__tr("This plugin will enable faxing support."),
  99.             PLUGIN_REASON_FAX_FEATURES: self.__tr("This plugin will enhnace faxing features."),
  100.             PLUGIN_REASON_RESERVED_20000: None,
  101.             PLUGIN_REASON_RESERVED_40000: None,
  102.         }
  103.  
  104.     #
  105.     # SOURCE PAGE
  106.     #
  107.     def showSourcePage(self):
  108.         reason_text = self.plugin_reason_text()
  109.  
  110.         if reason_text is not None:
  111.             if self.install_mode == PLUGIN_REQUIRED:
  112.                 self.TitleLabel.setText(self.__tr("An additional driver plug-in is required to operate this printer. You may download the plug-in directly from an HP authorized server (recommended), or, if you already have a copy of the file, you can specify a path to the file (advanced). <br><br>%1").arg(reason_text))
  113.                 self.SkipRadioButton.setEnabled(False)
  114.  
  115.             elif self.install_mode == PLUGIN_OPTIONAL:
  116.                 self.TitleLabel.setText(self.__tr("An optional driver plug-in is available to enhance the operation of this printer. You may download the plug-in directly from an HP authorized server (recommended), skip this installation (not recommended), or, if you already have a copy of the file, you can specify a path to the file (advanced).<br><br>%1").arg(reason_text))
  117.  
  118.         self.connect(self.DownloadRadioButton, SIGNAL("toggled(bool)"), self.DownloadRadioButton_toggled)
  119.         self.connect(self.CopyRadioButton, SIGNAL("toggled(bool)"), self.CopyRadioButton_toggled)
  120.         self.connect(self.SkipRadioButton, SIGNAL("toggled(bool)"), self.SkipRadioButton_toggled)
  121.         self.connect(self.PathLineEdit, SIGNAL("textChanged(const QString &)"), self.PathLineEdit_textChanged)
  122.         self.connect(self.BrowseToolButton, SIGNAL("clicked()"), self.BrowseToolButton_clicked)
  123.  
  124.         self.BrowseToolButton.setIcon(QIcon(load_pixmap('folder_open', '16x16')))
  125.  
  126.         self.displayPage(PAGE_SOURCE)
  127.  
  128.  
  129.     def DownloadRadioButton_toggled(self, b):
  130.         if b:
  131.             self.PathLineEdit.setEnabled(False)
  132.             self.BrowseToolButton.setEnabled(False)
  133.             self.NextButton.setEnabled(True)
  134.             try:
  135.                 self.PathLineEdit.setStyleSheet("")
  136.             except AttributeError:
  137.                 pass
  138.             self.plugin_path = None
  139.  
  140.  
  141.     def CopyRadioButton_toggled(self, b):
  142.         if b:
  143.             self.PathLineEdit.setEnabled(True)
  144.             self.BrowseToolButton.setEnabled(True)
  145.             self.plugin_path = unicode(self.PathLineEdit.text())
  146.             self.setPathIndicators()
  147.  
  148.  
  149.     def SkipRadioButton_toggled(self, b):
  150.         if b:
  151.             self.PathLineEdit.setEnabled(False)
  152.             self.BrowseToolButton.setEnabled(False)
  153.             self.NextButton.setEnabled(True)
  154.             try:
  155.                 self.PathLineEdit.setStyleSheet("")
  156.             except AttributeError:
  157.                 pass
  158.             self.plugin_path = None
  159.  
  160.  
  161.     def PathLineEdit_textChanged(self, t):
  162.         self.plugin_path = unicode(t)
  163.         self.setPathIndicators()
  164.  
  165.  
  166.     def setPathIndicators(self):
  167.         ok = True
  168.         if not self.plugin_path or (self.plugin_path and os.path.isdir(self.plugin_path)):
  169.             self.PathLineEdit.setToolTip(self.__tr("You must specify a path to the '%1' file.").arg(self.core.plugin_name))
  170.             ok = False
  171.         elif os.path.basename(self.plugin_path) != self.core.plugin_name:
  172.             self.PathLineEdit.setToolTip(self.__tr("The plugin filename must be '%1'.").arg(self.core.plugin_name))
  173.             ok = False
  174.  
  175.         if not ok:
  176.             try:
  177.                 self.PathLineEdit.setStyleSheet("background-color: yellow; ")
  178.             except AttributeError:
  179.                 pass
  180.             self.NextButton.setEnabled(False)
  181.         else:
  182.             try:
  183.                 self.PathLineEdit.setStyleSheet("")
  184.             except AttributeError:
  185.                 pass
  186.             self.NextButton.setEnabled(True)
  187.             self.PathLineEdit.setToolTip(QString(""))
  188.  
  189.  
  190.     def BrowseToolButton_clicked(self):
  191.         t = unicode(self.PathLineEdit.text())
  192.  
  193.         if not os.path.exists(t):
  194.             path = unicode(QFileDialog.getOpenFileName(self, self.__tr("Select Plug-in File"),
  195.                                                 #user_conf.workingDirectory(),
  196.                                                 self.user_settings.working_dir,
  197.                                                 self.__tr("Plugin Files (*.run)")))
  198.  
  199.         if path:
  200.             self.plugin_path = path
  201.             self.PathLineEdit.setText(self.plugin_path)
  202.             #user_conf.setWorkingDirectory(self.plugin_path)
  203.             self.user_settings.working_dir = self.plugin_path
  204.             self.user_settings.save()
  205.  
  206.         self.setPathIndicators()
  207.  
  208.     #
  209.     # Misc
  210.     #
  211.  
  212.     def displayPage(self, page):
  213.         self.updateStepText(page)
  214.         self.StackedWidget.setCurrentIndex(page)
  215.  
  216.  
  217.     def CancelButton_clicked(self):
  218.         self.close()
  219.  
  220.  
  221.     def NextButton_clicked(self):
  222.         if self.SkipRadioButton.isChecked():
  223.             log.debug("Skipping plug-in installation.")
  224.             self.close()
  225.             return
  226.  
  227.         beginWaitCursor()
  228.         try:
  229.  
  230.             if self.plugin_path is None: # download
  231.                 # read plugin.conf (local or on sf.net) to get plugin_path (http://)
  232.                 plugin_conf_url = self.core.get_plugin_conf_url()
  233.  
  234.                 if plugin_conf_url.startswith('file://'):
  235.                     pass
  236.                 else:
  237.                     log.info("Checking for network connection...")
  238.                     ok = self.core.check_network_connection()
  239.  
  240.                     if not ok:
  241.                         log.error("Network connection not detected.")
  242.                         endWaitCursor()
  243.                         FailureUI(self, self.__tr("Network connection not detected."))
  244.                         self.close()
  245.                         return
  246.  
  247.                 log.info("Downloading configuration file from: %s" % plugin_conf_url)
  248.                 self.plugin_path, size, checksum, timestamp, ok = self.core.get_plugin_info(plugin_conf_url,
  249.                     self.plugin_download_callback)
  250.  
  251.                 log.debug("path=%s, size=%d, checksum=%s, timestamp=%f, ok=%s" %
  252.                         (self.plugin_path, size, checksum, timestamp, ok))
  253.  
  254.                 if not self.plugin_path.startswith('http://') and not self.plugin_path.startswith('file://'):
  255.                     self.plugin_path = 'file://' + self.plugin_path
  256.  
  257.             else: # path
  258.                 if not self.plugin_path.startswith('http://'):
  259.                     self.plugin_path = 'file://' + self.plugin_path
  260.  
  261.                 size, checksum, timestamp = 0, '', 0
  262.  
  263.             if self.plugin_path.startswith('file://'):
  264.                 pass
  265.             else:
  266.                 log.info("Checking for network connection...")
  267.                 ok = self.core.check_network_connection()
  268.  
  269.                 if not ok:
  270.                     log.error("Network connection not detected.")
  271.                     endWaitCursor()
  272.                     FailureUI(self, self.__tr("Network connection not detected."))
  273.                     self.close()
  274.                     return
  275.  
  276.             log.info("Downloading plug-in from: %s" % self.plugin_path)
  277.  
  278.             status, ret = self.core.download_plugin(self.plugin_path, size, checksum, timestamp,
  279.                 self.plugin_download_callback)
  280.  
  281.             if status in (PLUGIN_INSTALL_ERROR_UNABLE_TO_RECV_KEYS, PLUGIN_INSTALL_ERROR_DIGITAL_SIG_NOT_FOUND):
  282.                 endWaitCursor()
  283.                 if QMessageBox.question(self, self.__tr("Digital signature download failed"),
  284.                         self.__tr("<b>The download of the digital signature file failed.</b><p>Without this file, it is not possible to authenticate and validate the plug-in prior to installation.</p>Do you still want to install the plug-in?"),
  285.                         QMessageBox.Yes | QMessageBox.No) != QMessageBox.Yes:
  286.  
  287.                     self.core.delete_plugin()
  288.                     self.close()
  289.                     return
  290.  
  291.             elif status != PLUGIN_INSTALL_ERROR_NONE:
  292.  
  293.                 if status == PLUGIN_INSTALL_ERROR_PLUGIN_FILE_NOT_FOUND:
  294.                     desc = self.__tr("<b>ERROR: Plug-in file not found (server returned 404 or similar error).</b><p>Error code: %1</p>").arg(str(ret))
  295.  
  296.                 elif status == PLUGIN_INSTALL_ERROR_DIGITAL_SIG_BAD:
  297.                     desc = self.__tr("<b>ERROR: Plug-in file does not match its digital signature.</b><p>File may have been corrupted or altered.</p><p>Error code: %1</p>").arg(str(ret))
  298.  
  299.                 elif status == PLUGIN_INSTALL_ERROR_PLUGIN_FILE_CHECKSUM_ERROR:
  300.                     desc = self.__tr("<b>ERROR: Plug-in file does not match its checksum. File may have been corrupted or altered.")
  301.  
  302.                 elif status == PLUGIN_INSTALL_ERROR_NO_NETWORK:
  303.                     desc = self.__tr("<b>ERROR: Unable to connect to network to download the plug-in.</b><p>Please check your network connection and try again.</p>")
  304.  
  305.                 elif status == PLUGIN_INSTALL_ERROR_DIRECTORY_ERROR:
  306.                     desc = self.__tr("<b>ERROR: Unable to create the plug-in directory.</b><p>Please check your permissions and try again.</p>")
  307.  
  308.                 self.core.delete_plugin()
  309.                 endWaitCursor()
  310.                 FailureUI(self, desc)
  311.                 self.close()
  312.                 return
  313.  
  314.             if not self.core.run_plugin(GUI_MODE, self.plugin_install_callback):
  315.                 self.core.delete_plugin()
  316.                 endWaitCursor()
  317.                 FailureUI(self, self.__tr("Plug-in install failed."))
  318.                 self.close()
  319.                 return
  320.  
  321.             cups_devices = device.getSupportedCUPSDevices(['hp'])
  322.             for dev in cups_devices:
  323.                 mq = device.queryModelByURI(dev)
  324.  
  325.                 if mq.get('fw-download', False):
  326.                     # Download firmware if needed
  327.                     log.info(log.bold("\nDownloading firmware to device %s..." % dev))
  328.                     try:
  329.                         d = None
  330.                         try:
  331.                             d = device.Device(dev)
  332.                         except Error:
  333.                             log.error("Error opening device.")
  334.                             endWaitCursor()
  335.                             FailureUI(self, self.__tr("<b>Firmware download to device failed.</b><p>%1</p>").arg(dev))
  336.                             continue
  337.  
  338.                         if d.downloadFirmware():
  339.                             log.info("Firmware download successful.\n")
  340.                         else:
  341.                             endWaitCursor()
  342.                             FailureUI(self, self.__tr("<b>Firmware download to device failed.</b><p>%1</p>").arg(dev))
  343.  
  344.                     finally:
  345.                         if d is not None:
  346.                             d.close()
  347.         finally:
  348.             endWaitCursor()
  349.  
  350.         self.core.delete_plugin()
  351.         SuccessUI(self, self.__tr("<b>Plug-in installation successful.</b>"))
  352.         self.result = True
  353.         self.close()
  354.  
  355.  
  356.     def plugin_download_callback(self, c, s, t):
  357.         pass
  358.  
  359.  
  360.     def plugin_install_callback(self, s):
  361.         print s
  362.  
  363.  
  364.     def updateStepText(self, p):
  365.         self.StepText.setText(self.__tr("Step %1 of %2").arg(p+1).arg(PAGE_MAX+1))
  366.  
  367.  
  368.     def plugin_reason_text(self):
  369.         try:
  370.             return self.PLUGIN_REASON_TEXT[self.plugin_reason]
  371.         except KeyError:
  372.             return None
  373.  
  374.  
  375.     def __tr(self,s,c = None):
  376.         return qApp.translate("PluginDialog",s,c)
  377.  
  378.  
  379.